<template>
  <div class="upload-file">
    <el-upload
      multiple
      :action="uploadFileUrl"
      :data="data"
      :before-upload="handleBeforeUpload"
      :file-list="fileList"
      :limit="limit"
      :on-error="handleUploadError"
      :on-exceed="handleExceed"
      :on-success="handleUploadSuccess"
      :show-file-list="false"
      :headers="headers"
      class="upload-file-uploader"
      ref="fileUpload"
      drag
    >
      <!-- 上传按钮 -->
      <!--      <el-button size="mini" type="primary">选取文件</el-button>-->

      <i class="el-icon-upload"></i>
      <div class="el-upload__text">将文件拖到此处，或<em>点击上传</em></div>

      <!-- 上传提示 -->
      <div class="el-upload__tip" slot="tip" v-if="showTip">
        请上传
        <template v-if="fileSize"> 大小不超过 <b style="color: #f56c6c">{{ fileSize }}MB</b></template>
        <template v-if="fileType"> 格式为 <b style="color: #f56c6c">{{ fileType.join("/") }}</b></template>
        的文件
      </div>
    </el-upload>
  </div>
</template>

<script>
import {getAccessToken} from "@/utils/auth";
import MP4Box from 'mp4box'

export default {
  name: "VideoFileUpload",
  props: {
    // 值
    value: [String, Object, Array],
    // 数量限制
    limit: {
      type: Number,
      default: 5,
    },
    // 大小限制(MB)
    fileSize: {
      type: Number,
      default: 500,
    },
    // 文件类型, 例如['png', 'jpg', 'jpeg']
    fileType: {
      type: Array,
      default: () => ["mp4"],
    },
    // 是否显示提示
    isShowTip: {
      type: Boolean,
      default: true
    }
  },
  data() {
    return {
      number: 0,
      uploadList: [],
      baseUrl: process.env.VUE_APP_BASE_API,
      uploadFileUrl: process.env.VUE_APP_BASE_API + "/admin-api/audit/video-info/upload", // 请求地址
      headers: {Authorization: "Bearer " + getAccessToken()}, // 设置上传的请求头部
      data: {},
      fileList: [],
    };
  },
  watch: {
    value: {
      handler(val) {
        if (val) {
          let temp = 1;
          // 首先将值转为数组
          const list = Array.isArray(val) ? val : this.value.split(',');
          // 然后将数组转为对象数组
          this.fileList = list.map(item => {
            if (typeof item === "string") {
              item = {name: item, url: item};
            }
            item.uid = item.uid || new Date().getTime() + temp++;
            return item;
          });
        } else {
          this.fileList = [];
          return [];
        }
      },
      deep: true,
      immediate: true
    }
  },
  computed: {
    // 是否显示提示
    showTip() {
      return this.isShowTip && (this.fileType || this.fileSize);
    },
  },
  methods: {
    // 上传前校检格式和大小
    async handleBeforeUpload(file) {
      // 校检文件类型
      if (this.fileType) {
        let fileExtension = "";
        if (file.name.lastIndexOf(".") > -1) {
          fileExtension = file.name.slice(file.name.lastIndexOf(".") + 1);
        }
        const isTypeOk = this.fileType.some((type) => {
          if (file.type.indexOf(type) > -1) return true;
          if (fileExtension && fileExtension.indexOf(type) > -1) return true;
          return false;
        });
        if (!isTypeOk) {
          this.$modal.msgError(`文件格式不正确, 请上传${this.fileType.join("/")}格式文件!`);
          return false;
        }
      }
      // 校检文件大小
      if (this.fileSize) {
        const isLt = file.size / 1024 / 1024 < this.fileSize;
        if (!isLt) {
          this.$modal.msgError(`上传文件大小不能超过 ${this.fileSize} MB!`);
          return false;
        }
      }
      this.$modal.loading("正在上传文件，请稍候...");

      // 计算视频时长和分辨率
      try {
        const videoInfo = await this.getVideoInfo(file); // 使用替换后的 MP4Box.js 方法
        if (videoInfo) {
          this.data.videoDuration = videoInfo.duration;
          this.data.videoResolution = `${videoInfo.width}*${videoInfo.height}`;
          this.data.videoCodec = videoInfo.codec;
        } else {
          this.$modal.msgError("无法计算视频信息，请检查文件!");
          return false;
        }
      } catch (err) {
        console.error(err);
        this.$modal.msgError("出现错误，无法读取视频信息!");
        return false;
      }

      this.number++;
      return true;
    },
    // 文件个数超出
    handleExceed() {
      this.$modal.msgError(`上传文件数量不能超过 ${this.limit} 个!`);
    },
    // 上传失败
    handleUploadError(err) {
      this.$modal.msgError("视频上传失败，请重试");
      this.$modal.closeLoading()
    },
    // 上传成功回调
    handleUploadSuccess(res, file) {
      if (res.code === 0) {
        // edit by 芋道源码
        this.uploadList.push({name: res.data, url: res.data});
        this.uploadedSuccessfully();
      } else {
        this.number--;
        this.$modal.closeLoading();
        this.$modal.msgError(res.msg);
        this.$refs.fileUpload.handleRemove(file);
        this.uploadedSuccessfully();
      }
    },
    // 删除文件
    handleDelete(index) {
      this.fileList.splice(index, 1);
      this.$emit("input", this.listToString(this.fileList));
    },
    // 上传结束处理
    uploadedSuccessfully() {
      if (this.number > 0 && this.uploadList.length === this.number) {
        this.fileList = this.fileList.concat(this.uploadList);
        this.uploadList = [];
        this.number = 0;
        this.$emit("input", this.listToString(this.fileList));
        this.$modal.closeLoading();
      }
    },
    // 获取文件名称
    getFileName(name) {
      if (name.lastIndexOf("/") > -1) {
        return name.slice(name.lastIndexOf("/") + 1);
      } else {
        return "";
      }
    },
    // 对象转成指定字符串分隔
    listToString(list, separator) {
      let strs = "";
      separator = separator || ",";
      for (let i in list) {
        strs += list[i].url + separator;
      }
      return strs != '' ? strs.substr(0, strs.length - 1) : '';
    },
    // 使用 MP4Box.js 获取视频分辨率、编码格式和时长
    async getVideoInfo(file) {
      return new Promise((resolve, reject) => {
        const fileReader = new FileReader();

        fileReader.onload = (event) => {
          try {
            const fileArrayBuffer = event.target.result;
            const mp4boxFile = MP4Box.createFile(); // 创建 MP4Box 文件实例
            mp4boxFile.onError = (e) => reject("MP4Box 解析错误：" + e);
            mp4boxFile.onReady = (info) => {
              // 视频轨道信息位于 `tracks`
              const videoTrack = info.tracks[0];
              console.log(videoTrack);
              if (videoTrack) {
                resolve({
                  width: videoTrack.video ? videoTrack.video.width : videoTrack.track_width,       // 视频宽度
                  height: videoTrack.video ? videoTrack.video.height : videoTrack.track_height,     // 视频高度
                  duration: videoTrack.movie_duration,      // 时长（毫秒）
                  codec: videoTrack.codec,             // 编码格式
                  size: videoTrack.size,               // 视频轨道大小
                });
              } else {
                reject("未找到视频轨道信息！");
              }
            };

            // 将 ArrayBuffer 加入 MP4Box 解析
            const arrayBufferWithBox = fileArrayBuffer.slice(0);
            arrayBufferWithBox.fileStart = 0;
            mp4boxFile.appendBuffer(arrayBufferWithBox);
            mp4boxFile.flush(); // 完成解析
          } catch (error) {
            reject("MP4Box.js 获取视频信息失败：" + error.message);
          }
        };

        fileReader.onerror = () => {
          reject("无法读取视频文件，请重试！");
        };

        // 将文件读取成 ArrayBuffer
        fileReader.readAsArrayBuffer(file);
      });
    }
  }
};
</script>

<style scoped lang="scss">
.upload-file-uploader {
  margin-bottom: 5px;
}

.upload-file-list .el-upload-list__item {
  border: 1px solid #e4e7ed;
  line-height: 2;
  margin-bottom: 10px;
  position: relative;
}

.upload-file-list .ele-upload-list__item-content {
  display: flex;
  justify-content: space-between;
  align-items: center;
  color: inherit;
}

.ele-upload-list__item-content-action .el-link {
  margin-right: 10px;
}
</style>
